package database

import (
	"fmt"
	"log"

	"gitee.com/xchengli/blog-lib/config"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

var DB *gorm.DB // GORM数据库实例

// Init 初始化GORM数据库连接
func Init() {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=True&loc=Local",
		config.DbCfg.DbUser,
		config.DbCfg.DbPass,
		config.DbCfg.DbHost,
		config.DbCfg.DbPort,
		config.DbCfg.DbName,
	)

	var err error
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	})
	if err != nil {
		log.Fatal("Failed to connect to database:", err)
	}

	// 获取原生SQL数据库连接以设置连接池参数
	sqlDB, err := DB.DB()
	if err != nil {
		log.Fatal("Failed to get SQL DB:", err)
	}

	// 设置连接池参数
	sqlDB.SetMaxOpenConns(100)
	sqlDB.SetMaxIdleConns(10)

	log.Println("Database connected successfully")
}

// GetDB 获取GORM数据库实例
func GetDB() *gorm.DB {
	return DB
}

// Close 关闭数据库连接
func Close() {
	// 关闭GORM数据库连接
	if DB != nil {
		sqlDB, err := DB.DB()
		if err == nil {
			sqlDB.Close()
			log.Println("Database connection closed")
		}
	}

	// 关闭Redis连接
	if RedisConn != nil {
		RedisConn.Close()
		log.Println("Redis connection closed")
	}
}
